# !usr/bin/env python
# -*- coding:utf-8 _*-
"""
@Author:张广勤
@Web site: https://www.tunan.wang
@Github:www.github.com
 
@File:docx_keywords1_0.py
@Time:2024/8/28 14:12

@Motto:不积跬步无以至千里，不积小流无以成江海！
"""
from docx import Document

def search_and_extract_paragraphs(folder_path, keyword, output_docx):
    # 创建一个新的Word文档用于写入结果
    doc = Document()

    # 遍历文件夹中的所有文件
    for filename in os.listdir(folder_path):
        if filename.endswith('.docx'):
            file_path = os.path.join(folder_path, filename)
            # 读取Word文件
            doc_to_read = Document(file_path)
            # 遍历文档中的所有段落
            for para in doc_to_read.paragraphs:
                if keyword in para.text:
                    # 如果段落包含关键字，则添加到新文档中，并在前面添加文件名
                    new_para = doc.add_paragraph(f"文件: {filename}\n{para.text}")

    # 保存新文档
    doc.save(output_docx)

# 使用示例
folder_path = '全国统计公报/'  # 替换为你的Word文件所在的文件夹路径
keyword = '装机容量'  # 指定要搜索的关键字
output_docx = 'output_with_'+keyword+'.docx'  # 输出文件的名称

# 确保导入了os模块，用于处理文件路径
import os

# 调用函数
search_and_extract_paragraphs(folder_path, keyword, output_docx)